Python中函数ravel()的说明:

您所在的位置:网站首页 numpy array ravel Python中函数ravel()的说明:

Python中函数ravel()的说明:

2023-08-31 15:19| 来源: 网络整理| 查看: 265

Python中函数ravel()的说明:

一开始是遇到:

predictions.append(pred.detach().cpu().numpy().ravel()[0]) # ravel展平, [0]取数值

单独运行发现没有ravel这个函数?而且没找到帮助文档?后来发现在numpy里面有这个函数 ravel的说明文档: 在这里插入图片描述 mgrid的说明文档: 在这里插入图片描述 看文档不够直观,直接看实例,reshape(-1)等同于这个ravel(): 在这里插入图片描述

代码:

import numpy as np a, b = np.mgrid[1:4:1, 2:5:1] print(a.shape) print(b.shape) print(a, b) a_ravel = a.ravel() print(a_ravel) b_ravel = a.ravel() print(b_ravel) a_reshape = a.reshape(-1) print(a_reshape)

经常在torch里面使用的view也有类似的打平操作 在这里插入图片描述 代码:

import torch x = torch.randn(2, 3) print(x.shape) print(x) x.view(1, -1) print(x.shape) 参考链接:

https://blog.csdn.net/yunfeather/article/details/106316811



【本文地址】


今日新闻


推荐新闻


CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3